package com.tt.common.util;

import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import java.util.ResourceBundle;

/**
 * Properties文件的工具类
 * ResourceBundle类是用来读取properties的资源文件的，很多国际化操作都使用该类。
 * @author ShangJinyao
 *
 */
@SuppressWarnings("all")
public class PropertiesUtils 
{
	/**
	 * 获得指定key的value值
	 * 如果是读取的文件带有“properties”后缀名，它会从工程根目录下找
	 * 如果是到classes下则去掉后缀
	 * @param fileName 资源文件名称
	 * @param key key名称
	 * @return
	 */
	public static String getKey_val(String fileName, String keyName) 
	{
		ResourceBundle cache = ResourceBundle.getBundle(fileName);
		return cache.getString(keyName);
	}
	
	/**
	 * 读取properties的全部信息
	 * @param filePath 文件路径
	 * @return
	 */
	public static Map<String, String> readProperties(String filePath) 
	{
		Properties props = new Properties();
		Map<String, String> map = new HashMap<String, String>();
		try
		{
			InputStream in = new BufferedInputStream(new FileInputStream(filePath));
			props.load(in);
			Enumeration en = props.propertyNames();
			while (en.hasMoreElements()) 
			{
				String key = (String) en.nextElement();
				String Property = props.getProperty(key);
				map.put(key, Property);
			}
		} catch (Exception e) 
		{
			e.printStackTrace();
		}
		return map;
	}
	
	/**
	 * 写入properties信息
	 * @param filePath 文件路径
	 * @param parameterName 属性名
	 * @param parameterValue 属性值
	 */
	public static void writeProperties(String filePath, String parameterName, String parameterValue)
	{
		Properties prop = new Properties();
		try 
		{
			InputStream fis = new FileInputStream(filePath);
			
			// 从输入流中读取属性列表（键和元素对）
			prop.load(fis);
			
			// 调用 Hashtable 的方法 put。使用 getProperty 方法提供并行性。
			// 强制要求为属性的键和值使用字符串。返回值是 Hashtable 调用 put 的结果。
			OutputStream fos = new FileOutputStream(filePath);
			prop.setProperty(parameterName, parameterValue);
			
			// 以适合使用 load 方法加载到 Properties 表中的格式，
			// 将此 Properties 表中的属性列表（键和元素对）写入输出流
			prop.store(fos, "");
		} catch (IOException e) 
		{
			System.err.println("Visit " + filePath + " for updating " + parameterName + " value error");
		}
	}
	
}
